# -*- coding:utf-8 -*-
# @Time: 2020/9/15 17:34
# @Author: 周铖鑫
# @Description：设置figure的坐标轴
# @File: SetAxis.py
import  matplotlib.pyplot as plt
import numpy as np
x=np.linspace(-3,3,50)
y1=x*2+1
y2=x**2
plt.figure(num=2,figsize=(10,4))
#放置两条线
plt.plot(x,y2)
plt.plot(x,y1,color='red',linewidth=1.0,linestyle='--')
#设置坐标轴范围
plt.xlim((-1,2))
plt.ylim((-2,3))
#设置坐标轴标签
plt.xlabel('I am x')
plt.ylabel('I am y')
new_ticks=np.linspace(-1,2,5)
print(new_ticks)
plt.xticks(new_ticks)
plt.yticks([-2,-1.8,-1,1.22,3],
            [r'$really\ bad$','$bad\ \\alpha$','$normal$','$good$','$really\ good$'])
#gca='get current axis'
ax=plt.gca()
#spines上下左右边框
ax.spines['right'].set_color('none')
ax.spines['top'].set_color('none')
ax.xaxis.set_ticks_position('bottom')
ax.yaxis.set_ticks_position('left')
ax.spines['bottom'].set_position(('data',0))
ax.spines['left'].set_position(('data',0))
plt.show()